package org.bonorsoft.modules.control.util;

import com.google.common.collect.Lists;
import org.apache.commons.io.IOUtils;
import org.bonorsoft.modules.control.entity.Address;
import org.bonorsoft.modules.control.service.IAddressService;
import org.springframework.beans.factory.annotation.Autowired;
import java.io.*;
import java.nio.file.Files;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

public class ZipFileUtil {
    @Autowired
    private IAddressService addressService;

    public void qrcodeZip(String ids, OutputStream outputStream) {
        //我这里的ids是实体对象的id用英文逗号分隔拼成的字符串,
        //到业务实现类里面拿着去数据层取对象生成文件的,根据你们需求不同可以改
        ZipOutputStream zipOutStream = null;
        //------走完for循环得到一个文件集合就行-----
        List<File> fileList = Lists.newArrayList();
        for (String id : ids.split(",")) {
            Address address = addressService.getById(id);
            String dzmc = address.getDzmc();
            //------文件生成并给出路径-----
            String filePath = ("C:\\police\\upFiles\\qrcode\\" + dzmc + ".jpg");
            File file = new File(filePath);
            fileList.add(file);
        }
        try {
            //-- 包装成ZIP格式输出流
            zipOutStream = new ZipOutputStream(new BufferedOutputStream(outputStream));
            // -- 设置压缩方法
            zipOutStream.setMethod(ZipOutputStream.DEFLATED);
            //-- 将多文件循环写入压缩包
            for (File file : fileList) {
                byte[] data = Files.readAllBytes(file.toPath());
                zipOutStream.putNextEntry(new ZipEntry(file.getName()));
                zipOutStream.write(data);
                zipOutStream.closeEntry();
                zipOutStream.flush();
            }
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            IOUtils.closeQuietly(zipOutStream);
            IOUtils.closeQuietly(outputStream);
        }
    }

    /**
     * 将存放在sourceFilePath目录下的源文件,打包成fileName名称的ZIP文件,并存放到zipFilePath。
     *
     * @param sourceFilePath 待压缩的文件路径
     * @param zipFilePath    压缩后存放路径
     * @param fileName       压缩后文件的名称
     * @return flag
     */
    public static boolean fileToZip(String sourceFilePath, String zipFilePath, String fileName) {
        boolean flag = false;
        File sourceFile = new File(sourceFilePath);
        FileInputStream fis = null;
        BufferedInputStream bis = null;
        FileOutputStream fos = null;
        ZipOutputStream zos = null;

        if (sourceFile.exists() == false) {
            System.out.println(">>>>>> 待压缩的文件目录：" + sourceFilePath + " 不存在. <<<<<<");
        } else {
            try {
                File zipFile = new File(zipFilePath + "/" + fileName + ".zip");
                if (zipFile.exists()) {
                    System.out.println(">>>>>> " + zipFilePath + " 目录下存在名字为："
                            + fileName + ".zip" + " 打包文件. <<<<<<");
                } else {
                    File[] sourceFiles = sourceFile.listFiles();
                    if (null == sourceFiles || sourceFiles.length < 1) {
                        System.out.println(">>>>>> 待压缩的文件目录：" + sourceFilePath
                                + " 里面不存在文件,无需压缩. <<<<<<");
                    } else {
                        fos = new FileOutputStream(zipFile);
                        zos = new ZipOutputStream(new BufferedOutputStream(fos));
                        byte[] bufs = new byte[1024 * 10];
                        for (int i = 0; i < sourceFiles.length; i++) {
                            // 创建ZIP实体,并添加进压缩包
                            ZipEntry zipEntry = new ZipEntry(sourceFiles[i].getName());
                            zos.putNextEntry(zipEntry);
                            // 读取待压缩的文件并写进压缩包里
                            fis = new FileInputStream(sourceFiles[i]);
                            bis = new BufferedInputStream(fis, 1024 * 10);
                            int read = 0;
                            while ((read = bis.read(bufs, 0, 1024 * 10)) != -1) {
                                zos.write(bufs, 0, read);
                            }
                        }
                        flag = true;
                    }
                }
            } catch (FileNotFoundException e) {
                e.printStackTrace();
                throw new RuntimeException(e);
            } catch (IOException e) {
                e.printStackTrace();
                throw new RuntimeException(e);
            } finally {
                // 关闭流
                try {
                    if (null != bis) {
                        bis.close();
                    }
                    if (null != zos) {
                        zos.close();
                    }
                } catch (IOException e) {
                    e.printStackTrace();
                    throw new RuntimeException(e);
                }
            }
        }
        return flag;
    }


    /**
     * 将文件打包成ZIP压缩文件,main方法测试
     *
     * @param args
     */

    public static void main(String[] args) {
        String sourceFilePath = "E:\\perTest\\UserMsg";
        String zipFilePath = "E:\\perTest";
        String fileName = "ZIP";
        boolean flag = ZipFileUtil.fileToZip(sourceFilePath, zipFilePath,
                fileName);
        if (flag) {
            System.out.println(">>>>>> 文件打包成功. <<<<<<");
        } else {
            System.out.println(">>>>>> 文件打包失败. <<<<<<");
        }
    }
}
